cd ../..

gpus=$1
psd_lo_thd=0.10
psd_hi_thd=0.90
weight=70
python -m trainer.ATSTContrastiveTrainer test \
    --config configs/model/ATSTSED_contrastive.yaml \
    --config configs/dataset/desed_with_id.yaml \
    --config configs/trainer/default.yaml \
    --config configs/experiment/intplcontrastive.yaml \
    --config configs/experiment/plugin_ATSTFrame.yaml \
    --model.exp_name="5_cl_usage/ATSTSED/+wCL_${weight}/mask_both/" \
    --data.init_args.batch_sizes=[12,12,12,192] \
    --model.cl_decay=True \
    --model.cl_weight=${weight} \
    --trainer.devices=${gpus} \
    --trainer.strategy="ddp_find_unused_parameters_true" \
    --model.ckpt_path="/home/shaonian/sed/logs/5_cl_usage/ATSTSED/stage2/+wCL_0_wMT_70/prev/version_7/checkpoints/epoch59_metric0.0884.ckpt" \
    --model.psd_label_filter=False \
    --model.psd_lo_thd=${psd_lo_thd} \
    --model.psd_hi_thd=${psd_hi_thd} \
    --model.unsupervised_loss.init_args.weight=2 \
    --data.init_args.val_batch_size=1 
# done